Fix for a stupid bug :)