just a little bug fixing