bug fixing on test