Fixed bug introduced last commit