Fixes to module a_star

The type error was detected by nitg in complex cases.