bug fix around implementation for ui