Fixed small bug, should work now