little bug fixing