Fixed a small bug with the new setup :P