Fixed small problem which was not supposed to have been commited