Fixed two use statements