Minor improvement added on Repository class