Fixed small bug, late night ;)