bug/error fixes