fixing one last issue