FIX minor bug