Bug fix - use statement