fixed bug for testing