diff --git a/flake.nix b/flake.nix
index 3f59074..5cda8c7 100644
--- a/flake.nix
+++ b/flake.nix
@@ -2,8 +2,7 @@
   description = "Testing mastodon configurations";
 
   inputs = {
-    nixpkgs.url = "github:nixos/nixpkgs?ref=nixos-unstable";
-    # nixpkgs.url = "path:/home/qolen/nixpkgs";
+    nixpkgs.url = "github:radvendii/nixpkgs/nixos_rebuild_tests";
   };
 
   outputs = { self, nixpkgs }: 
diff --git a/tests/mastodon-garage.nix b/tests/mastodon-garage.nix
index f99d572..22605b8 100644
--- a/tests/mastodon-garage.nix
+++ b/tests/mastodon-garage.nix
@@ -1,7 +1,6 @@
 { pkgs, self }:
 let
   lib = pkgs.lib;
-  # python = pkgs.python310.withPackages (ps: with ps; [ requests aiokafka ]);
   rebuildableTest = import ./rebuildableTest.nix pkgs;
   seleniumScript = pkgs.writers.writePython3Bin "selenium-script"
     {
@@ -33,7 +32,7 @@ let
     driver.close()
   '';
 in
-rebuildableTest {
+pkgs.nixosTest {
   name = "test-mastodon-garage";
 
   nodes = {