fix errors after testing