Small modification suggested from hlint