forgot one little piece to the functionality