Fix incorrect import and javadoc