Fixed small bug preventing me from testing