BUG/TST: Add check for type of table