fixing warning, ready for second milestone